from flask import Blueprint
from flask import session, redirect, request, url_for

admin_blue = Blueprint('admin_blue', __name__, url_prefix='/admin')

from . import views


# 请求钩子，检查用户登录权限
@admin_blue.before_request
def check_admin():
    # 从session中获取管理员权限，如果没有获取到则不是管理员
    is_admin = session.get('is_admin', False)
    # 不是管理员或者访问的不是管理员登录界面，返回主界面
    if not is_admin and not request.url.endswith(url_for('admin_blue.admin_login')):
        return redirect('/')